#pragma once

#include <iostream>
#include <string>
#include <vector>
#include <semaphore.h>
#include <pthread.h>

template <typename T>
class RingQueue
{
public:
    // 申请资源
    void P(sem_t &s)
    {
        sem_wait(&s);
    }
    // 发布资源
    void V(sem_t &s)
    {
        sem_post(&s);
    }

public:
    RingQueue(const int max_cap)
        : _max_cap(max_cap), _ringqueue(max_cap), _c_step(0), _p_step(0)
    {
        // 初始化信号量
        sem_init(&_data_sem, 0, 0);
        sem_init(&_space_sem, 0, _max_cap);

        // 初始化锁
        pthread_mutex_init(&_c_mutex, nullptr);
        pthread_mutex_init(&_p_mutex, nullptr);

    }
    void Push(const T &in) // 生产者
    {
        P(_space_sem); // 申请空间资源
        pthread_mutex_lock(&_c_mutex);
        _ringqueue[_p_step] = in; // 把资源放进去
        _p_step++;
        _p_step %= _max_cap;
        pthread_mutex_unlock(&_c_mutex);
        V(_data_sem); // 把生产的资源发布出去
    }
    void Pop(T * out) // 消费者
    {
        P(_data_sem); // 申请数据资源
        pthread_mutex_lock(&_p_mutex);
        *out = _ringqueue[_c_step]; // 把资源取出来
        _c_step++;
        _c_step %= _max_cap;
        pthread_mutex_unlock(&_p_mutex);
        V(_space_sem); // 把空间资源还回去
    }

    ~RingQueue()
    {
        // 销毁信号量
        sem_destroy(&_data_sem);
        sem_destroy(&_space_sem);

        // 销毁锁
        pthread_mutex_destroy(&_c_mutex);
        pthread_mutex_destroy(&_p_mutex);

    }

private:
    std::vector<T> _ringqueue;
    int _max_cap; // 环形队列的容量

    int _c_step; // 消费者的位置
    int _p_step; // 生产者的位置

    sem_t _data_sem;  // 消费者关系
    sem_t _space_sem; // 生产者关系

    pthread_mutex_t _c_mutex;
    pthread_mutex_t _p_mutex; 

};